package com.fsck.k9.mail.internet; import java.util.Date; import java.util.Locale; import java.util.TimeZone; import android.test.AndroidTestCase; import com.fsck.k9.mail.Address; import com.fsck.k9.mail.MessagingException; import com.fsck.k9.mail.Message.RecipientType; import com.fsck.k9.mail.internet.MimeUtility.ViewableContainer; public class ViewablesTest extends AndroidTestCase { public void testSimplePlainTextMessage() throws MessagingException { String bodyText = "K-9 Mail rocks :>"; // Create text/plain body TextBody body = new TextBody(bodyText); // Create message MimeMessage message = new MimeMessage(); message.setBody(body); // Extract text ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message); String expectedText = bodyText; String expectedHtml = "<html><head/><body>" + "<pre style=\"white-space: pre-wrap; word-wrap:break-word; " + "font-family: sans-serif\">" + "K-9 Mail rocks :>" + "</pre>" + "</body></html>"; assertEquals(expectedText, container.text); assertEquals(expectedHtml, container.html); } public void testSimpleHtmlMessage() throws MessagingException { String bodyText = "<strong>K-9 Mail</strong> rocks :>"; // Create text/plain body TextBody body = new TextBody(bodyText); // Create message MimeMessage message = new MimeMessage(); message.setHeader("Content-Type", "text/html"); message.setBody(body); // Extract text ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message); String expectedText = "K-9 Mail rocks :>"; String expectedHtml = "<html><head/><body>" + bodyText + "</body></html>"; assertEquals(expectedText, container.text); assertEquals(expectedHtml, container.html); } public void testMultipartPlainTextMessage() throws MessagingException { String bodyText1 = "text body 1"; String bodyText2 = "text body 2"; // Create text/plain bodies TextBody body1 = new TextBody(bodyText1); TextBody body2 = new TextBody(bodyText2); // Create multipart/mixed part MimeMultipart multipart = new MimeMultipart(); MimeBodyPart bodyPart1 = new MimeBodyPart(body1, "text/plain"); MimeBodyPart bodyPart2 = new MimeBodyPart(body2, "text/plain"); multipart.addBodyPart(bodyPart1); multipart.addBodyPart(bodyPart2); // Create message MimeMessage message = new MimeMessage(); message.setBody(multipart); // Extract text ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message); String expectedText = bodyText1 + "\n\n" + "------------------------------------------------------------------------\n\n" + bodyText2; String expectedHtml = "<html><head/><body>" + "<pre style=\"white-space: pre-wrap; word-wrap:break-word; " + "font-family: sans-serif\">" + bodyText1 + "</pre>" + "<p style=\"margin-top: 2.5em; margin-bottom: 1em; " + "border-bottom: 1px solid #000\"></p>" + "<pre style=\"white-space: pre-wrap; word-wrap:break-word; " + "font-family: sans-serif\">" + bodyText2 + "</pre>" + "</body></html>"; assertEquals(expectedText, container.text); assertEquals(expectedHtml, container.html); } public void testTextPlusRfc822Message() throws MessagingException { Locale.setDefault(Locale.US); TimeZone.setDefault(TimeZone.getTimeZone("UTC")); String bodyText = "Some text here"; String innerBodyText = "Hey there. I'm inside a message/rfc822 (inline) attachment."; // Create text/plain body TextBody textBody = new TextBody(bodyText); // Create inner text/plain body TextBody innerBody = new TextBody(innerBodyText); // Create message/rfc822 body MimeMessage innerMessage = new MimeMessage(); innerMessage.addSentDate(new Date(112, 02, 17)); innerMessage.setRecipients(RecipientType.TO, new Address[] { new Address("to@example.com") }); innerMessage.setSubject("Subject"); innerMessage.setFrom(new Address("from@example.com")); innerMessage.setBody(innerBody); // Create multipart/mixed part MimeMultipart multipart = new MimeMultipart(); MimeBodyPart bodyPart1 = new MimeBodyPart(textBody, "text/plain"); MimeBodyPart bodyPart2 = new MimeBodyPart(innerMessage, "message/rfc822"); bodyPart2.setHeader("Content-Disposition", "inline; filename=\"message.eml\""); multipart.addBodyPart(bodyPart1); multipart.addBodyPart(bodyPart2); // Create message MimeMessage message = new MimeMessage(); message.setBody(multipart); // Extract text ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message); String expectedText = bodyText + "\n\n" + "----- message.eml ------------------------------------------------------" + "\n\n" + "From: from@example.com" + "\n" + "To: to@example.com" + "\n" + "Sent: Sat Mar 17 00:00:00 GMT+00:00 2012" + "\n" + "Subject: Subject" + "\n" + "\n" + innerBodyText; String expectedHtml = "<html><head/><body>" + "<pre style=\"white-space: pre-wrap; word-wrap:break-word; " + "font-family: sans-serif\">" + bodyText + "</pre>" + "<p style=\"margin-top: 2.5em; margin-bottom: 1em; border-bottom: " + "1px solid #000\">message.eml</p>" + "<table style=\"border: 0\">" + "<tr>" + "<th style=\"text-align: left; vertical-align: top;\">From:</th>" + "<td>from@example.com</td>" + "</tr><tr>" + "<th style=\"text-align: left; vertical-align: top;\">To:</th>" + "<td>to@example.com</td>" + "</tr><tr>" + "<th style=\"text-align: left; vertical-align: top;\">Sent:</th>" + "<td>Sat Mar 17 00:00:00 GMT+00:00 2012</td>" + "</tr><tr>" + "<th style=\"text-align: left; vertical-align: top;\">Subject:</th>" + "<td>Subject</td>" + "</tr>" + "</table>" + "<pre style=\"white-space: pre-wrap; word-wrap:break-word; " + "font-family: sans-serif\">" + innerBodyText + "</pre>" + "</body></html>"; assertEquals(expectedText, container.text); assertEquals(expectedHtml, container.html); } }